from tensorflow.examples.tutorials.mnist import input_data
import pylab

mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
print("input data", mnist.train.images)
print("input data's shape", mnist.train.images.shape)

im = mnist.train.images[1]
im = im.reshape(-1, 28)
pylab.imshow(im)
pylab.show()
